import os
from PyPDF2 import PdfReader, PdfWriter

# PDF文件名
pdf_filename = "合同.pdf"
# 输出目录名
output_dir = "合同目录"

# 确保输出目录存在，如果不存在则创建
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

# 打开PDF文件（文件要放在代码同级目录！）
reader = PdfReader(pdf_filename)

# 遍历PDF中的每一页
for i, page in enumerate(reader.pages):
    # 创建一个新的PdfWriter对象来写入每一页
    writer = PdfWriter()
    # 添加当前页到writer对象
    writer.add_page(page)
    # 构造输出文件的完整路径
    output_filename = f"合同_第{i+1}页.pdf"
    output_path = os.path.join(output_dir, output_filename)
    # 保存每一页到单独的文件
    with open(output_path, "wb") as f:
        writer.write(f)
    print(f"第{i+1}页拆分成功，文件保存至：{output_path}")

